Fix odd bug