Feature small fix

5 minutes, not 10