Light Sleeper

Hi mummies is it normal for 3months baby actually coming to 4 months in a week time will have his nap only about 30 mins. He will sleep longer if i held him but when i put him down he will wake up. Tried to put him in yaolan swing hoping that he can sleep longer but still the same. Any mummies experience the same thing? Will it get better as time goes?