Bug fixes :) :)