Add ugly hack