Adding the required parameter and returning to prevent fatal error

Fixes #2