Fixed mini-bug