Fix layout

Signed-off-by: Thanh Ha <thanh.ha@alumni.carleton.ca>