fixing bug with custom design