Fixed invalid return type