diff --git a/.changeset/polite-terms-dress.md b/.changeset/polite-terms-dress.md
new file mode 100644
index 0000000..de0690f
--- /dev/null
+++ b/.changeset/polite-terms-dress.md
@@ -0,0 +1,5 @@
+---
+'atomsphere': patch
+---
+
+Improve types in `atom` function
diff --git a/src/atom.ts b/src/atom.ts
index 631c9e3..6fe4a94 100644
--- a/src/atom.ts
+++ b/src/atom.ts
@@ -11,7 +11,9 @@ type DeriveFn<T> = (get: <P>(atom: ReadableAtom<P>) => P) => T;
 
 export function atom<T>(initialValue: DeriveFn<T>): ReadableAtom<T>;
 export function atom<T>(initialValue: T): WritableAtom<T>;
-export function atom<T>(initialValue: T | DeriveFn<T>) {
+export function atom<T>(
+	initialValue: T | DeriveFn<T>,
+): WritableAtom<T> | ReadableAtom<T> {
 	return typeof initialValue === 'function'
 		? createDerivedAtom(initialValue as DeriveFn<T>)
 		: createAtom(initialValue);