Hopefully bug free now :)

Signed-off-by: Levi Gross <levi@levigross.com>