Fixed error on work in progress