Fix mini-bug