little fix with return type of method