Fixed return value

Added missing return