little visibility fix
try to fix undefined method returning