Fixing a bug

Signed-off-by: Yuri Tomanek <yuri@tomanek.com.au>