Fixed problem with flow layout