Gotta fix the logic...

forgot to commit this...

Signed-off-by: Suki <suki@missallsunday.com>