Fix bug (ain't js great?)