Fixing #2

Forgot to solve the main problem asked