Fixed broken mock up