Add missing return from onClick