Err, fix stupid bug