Fix for mock method feature check